Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_telemetry_integration.py

Displaying Raw • Download

Tff7b72import T7ee787pytest

Tff7b72import T7ee787time

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMock

Tff7b72from T7ee787meshchatxT7ee787.T7ee787meshchat Tff7b72import Te6edf3ReticulumMeshChat

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787telemetry_utils Tff7b72import Te6edf3Telemeter

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8ffmock_appTb4b4b4(Tb4b4b4)Tb4b4b4:

T8b949e# We create a simple mock object that has the methods/attributes

T8b949e# needed by process_incoming_telemetry and other telemetry logic.

Te6edf3app Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3specTff7b72=Te6edf3ReticulumMeshChatTb4b4b4)

T8b949e# Mock database

Te6edf3appTff7b72.Td2a8ffdatabase Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3appTff7b72.Td2a8ffdatabaseTff7b72.Td2a8fftelemetry Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

T8b949e# Mock context

Te6edf3appTff7b72.Td2a8ffcurrent_context Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffdatabase Tff7b72= Te6edf3appTff7b72.Td2a8ffdatabase

Te6edf3appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8fflocal_lxmf_destination Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8fflocal_lxmf_destinationTff7b72.Td2a8ffhexhash Tff7b72= Ta5d6ff"Ta5d6fflocal_hashTa5d6ff"

T8b949e# Mock reticulum

Te6edf3appTff7b72.Td2a8ffreticulum Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3appTff7b72.Td2a8ffreticulumTff7b72.Td2a8ffget_packet_rssiTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72-T79c0ff70

Te6edf3appTff7b72.Td2a8ffreticulumTff7b72.Td2a8ffget_packet_snrTff7b72.Td2a8ffreturn_value Tff7b72= T79c0ff12.5

Te6edf3appTff7b72.Td2a8ffreticulumTff7b72.Td2a8ffget_packet_qTff7b72.Td2a8ffreturn_value Tff7b72= T79c0ff85

T8b949e# Mock websocket_broadcast

Te6edf3appTff7b72.Td2a8ffwebsocket_broadcast Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

T8b949e# Attach the actual method we want to test if possible,

T8b949e# but since it's an instance method, we might need to bind it.

Te6edf3appTff7b72.Td2a8ffprocess_incoming_telemetry Tff7b72= Tb4b4b4(

Te6edf3ReticulumMeshChatTff7b72.Td2a8ffprocess_incoming_telemetryTff7b72.Td2a8ff__get__Tb4b4b4(Te6edf3appTb4b4b4, Te6edf3ReticulumMeshChatTb4b4b4)

Tb4b4b4)

Tff7b72return Te6edf3app

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_process_incoming_telemetry_singleTb4b4b4(Te6edf3mock_appTb4b4b4)Tb4b4b4:

Te6edf3source_hash Tff7b72= Ta5d6ff"Ta5d6ffsource_hashTa5d6ff"

Te6edf3location Tff7b72= Tb4b4b4{Ta5d6ff"Ta5d6fflatitudeTa5d6ff"Tb4b4b4: T79c0ff50.0Tb4b4b4, Ta5d6ff"Ta5d6fflongitudeTa5d6ff"Tb4b4b4: T79c0ff10.0Tb4b4b4}

Te6edf3packed_telemetry Tff7b72= Te6edf3TelemeterTff7b72.Td2a8ffpackTb4b4b4(Te6edf3locationTff7b72=Te6edf3locationTb4b4b4)

Te6edf3mock_lxmf_message Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_lxmf_messageTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffmsg_hashTa5d6ff"

Te6edf3mock_appTff7b72.Td2a8ffprocess_incoming_telemetryTb4b4b4(

Te6edf3source_hashTb4b4b4, Te6edf3packed_telemetryTb4b4b4, Te6edf3mock_lxmf_message

Tb4b4b4)

T8b949e# Verify database call

Te6edf3mock_appTff7b72.Td2a8ffdatabaseTff7b72.Td2a8fftelemetryTff7b72.Td2a8ffupsert_telemetryTff7b72.Td2a8ffassert_calledTb4b4b4(Tb4b4b4)

Te6edf3call_args Tff7b72= Te6edf3mock_appTff7b72.Td2a8ffdatabaseTff7b72.Td2a8fftelemetryTff7b72.Td2a8ffupsert_telemetryTff7b72.Td2a8ffcall_argsTb4b4b4[T79c0ff1Tb4b4b4]

Tff7b72assert Te6edf3call_argsTb4b4b4[Ta5d6ff"Ta5d6ffdestination_hashTa5d6ff"Tb4b4b4] Tff7b72== Te6edf3source_hash

Tff7b72assert Te6edf3call_argsTb4b4b4[Ta5d6ff"Ta5d6ffdataTa5d6ff"Tb4b4b4] Tff7b72== Te6edf3packed_telemetry

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_process_incoming_telemetry_streamTb4b4b4(Te6edf3mock_appTb4b4b4)Tb4b4b4:

T8b949e# This simulates receiving a telemetry stream (e.g. from Sideband collector)

Te6edf3entries Tff7b72= Tb4b4b4[

Tb4b4b4(

Ta5d6ff"Ta5d6ffpeer1Ta5d6ff"Tb4b4b4,

Tffa657intTb4b4b4(Te6edf3timeTff7b72.Td2a8fftimeTb4b4b4(Tb4b4b4)Tb4b4b4) Tff7b72- T79c0ff60Tb4b4b4,

Te6edf3TelemeterTff7b72.Td2a8ffpackTb4b4b4(Te6edf3locationTff7b72=Tb4b4b4{Ta5d6ff"Ta5d6fflatitudeTa5d6ff"Tb4b4b4: T79c0ff1.0Tb4b4b4, Ta5d6ff"Ta5d6fflongitudeTa5d6ff"Tb4b4b4: T79c0ff1.0Tb4b4b4}Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4,

Tb4b4b4(

Ta5d6ff"Ta5d6ffpeer2Ta5d6ff"Tb4b4b4,

Tffa657intTb4b4b4(Te6edf3timeTff7b72.Td2a8fftimeTb4b4b4(Tb4b4b4)Tb4b4b4)Tb4b4b4,

Te6edf3TelemeterTff7b72.Td2a8ffpackTb4b4b4(Te6edf3locationTff7b72=Tb4b4b4{Ta5d6ff"Ta5d6fflatitudeTa5d6ff"Tb4b4b4: T79c0ff2.0Tb4b4b4, Ta5d6ff"Ta5d6fflongitudeTa5d6ff"Tb4b4b4: T79c0ff2.0Tb4b4b4}Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4,

Tb4b4b4]

Te6edf3mock_lxmf_message Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_lxmf_messageTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffstream_msg_hashTa5d6ff"

T8b949e# We call it directly for each entry as process_incoming_telemetry is refactored

T8b949e# to handle single entries, and on_lxmf_delivery loops over streams.

Tff7b72for Te6edf3entry_sourceTb4b4b4, Te6edf3entry_timestampTb4b4b4, Te6edf3entry_data Tff7b72in Te6edf3entriesTb4b4b4:

Te6edf3mock_appTff7b72.Td2a8ffprocess_incoming_telemetryTb4b4b4(

Te6edf3entry_sourceTb4b4b4,

Te6edf3entry_dataTb4b4b4,

Te6edf3mock_lxmf_messageTb4b4b4,

Te6edf3timestamp_overrideTff7b72=Te6edf3entry_timestampTb4b4b4,

Tb4b4b4)

Tff7b72assert Te6edf3mock_appTff7b72.Td2a8ffdatabaseTff7b72.Td2a8fftelemetryTff7b72.Td2a8ffupsert_telemetryTff7b72.Td2a8ffcall_count Tff7b72== T79c0ff2

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_telemetry_request_parsingTb4b4b4(Te6edf3mock_appTb4b4b4)Tb4b4b4:

T8b949e# Test that on_lxmf_delivery correctly identifies telemetry requests

T8b949e# and calls handle_telemetry_request.

Te6edf3mock_lxmf_message Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

T8b949e# 0x01 is SidebandCommands.TELEMETRY_REQUEST

T8b949e# We mock get_fields to return a command request

Te6edf3mock_lxmf_messageTff7b72.Td2a8ffget_fieldsTff7b72.Td2a8ffreturn_value Tff7b72= Tb4b4b4{T79c0ff0x01Tb4b4b4: Tb4b4b4[Tb4b4b4{T79c0ff0x01Tb4b4b4: Tffa657intTb4b4b4(Te6edf3timeTff7b72.Td2a8fftimeTb4b4b4(Tb4b4b4)Tb4b4b4)Tb4b4b4}Tb4b4b4]Tb4b4b4}

Te6edf3mock_lxmf_messageTff7b72.Td2a8ffsource_hash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffsource_hash_bytesTa5d6ff"

Te6edf3mock_lxmf_messageTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffmsg_hashTa5d6ff"

Te6edf3mock_lxmf_messageTff7b72.Td2a8ffdestination_hash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffdest_hashTa5d6ff"

T8b949e# We need to mock handle_telemetry_request on the app

Te6edf3mock_appTff7b72.Td2a8ffhandle_telemetry_request Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

T8b949e# Bind on_lxmf_delivery

Te6edf3mock_appTff7b72.Td2a8ffon_lxmf_delivery Tff7b72= Te6edf3ReticulumMeshChatTff7b72.Td2a8ffon_lxmf_deliveryTff7b72.Td2a8ff__get__Tb4b4b4(

Te6edf3mock_appTb4b4b4, Te6edf3ReticulumMeshChat

Tb4b4b4)

T8b949e# Mocking dependencies

Te6edf3mock_appTff7b72.Td2a8ffis_destination_blockedTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72False

Te6edf3mock_appTff7b72.Td2a8ffcurrent_contextTff7b72.Td2a8ffconfigTff7b72.Td2a8fftelemetry_enabledTff7b72.Td2a8ffgetTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72True

Te6edf3mock_appTff7b72.Td2a8ffdatabaseTff7b72.Td2a8ffcontactsTff7b72.Td2a8ffget_contact_by_identity_hashTff7b72.Td2a8ffreturn_value Tff7b72= Tb4b4b4{

Ta5d6ff"Ta5d6ffis_telemetry_trustedTa5d6ff"Tb4b4b4: Tff7b72True

Tb4b4b4}

Te6edf3mock_appTff7b72.Td2a8ffdatabaseTff7b72.Td2a8ffmessagesTff7b72.Td2a8ffget_lxmf_message_by_hashTff7b72.Td2a8ffreturn_value Tff7b72= Tb4b4b4{Tb4b4b4} T8b949e# To avoid JSON error

T8b949e# Call it

Te6edf3mock_appTff7b72.Td2a8ffon_lxmf_deliveryTb4b4b4(Te6edf3mock_lxmf_messageTb4b4b4)

T8b949e# Verify handle_telemetry_request was called

Te6edf3mock_appTff7b72.Td2a8ffhandle_telemetry_requestTff7b72.Td2a8ffassert_called_withTb4b4b4(

Ta5d6ff"Ta5d6ff736f757263655f686173685f6279746573Ta5d6ff"

Tb4b4b4)

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_tracking_toggle_endpointTb4b4b4(Te6edf3mock_appTb4b4b4)Tb4b4b4:

T8b949e# Mock database responses

Te6edf3mock_appTff7b72.Td2a8ffdatabaseTff7b72.Td2a8fftelemetryTff7b72.Td2a8ffis_trackingTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72False

T8b949e# We can't easily test the web endpoint here without more setup,

T8b949e# but we can test the logic it calls if it was refactored into a method.

Tff7b72pass